simple array sum hackerrank solution in python